package com.briup.dataset.entity.extend;

import com.briup.dataset.entity.DataSet;

import java.util.ArrayList;
import java.util.List;

/**
 * 数据集扩展类
 */
public class DataSetEx extends DataSet {
    // 查看所有数据集需要下面数据

    // 该数据集下的分类数量
    private int classifyNum;
    // 该数据集下所有的图片实体数量
    private int entityNum;

    // 数据下的分类名称以及分类对应的图片数量(此代码不推荐使用,考虑到需要复用)
    //private Map<String,Integer> entityInClassify = new HashMap<>();
    private List<ClassifyEx> classifyExList = new ArrayList<>();

    // 数据集分辨率 此处没有用到
    // private String datasetRatio;



    public int getClassifyNum() {
        return classifyNum;
    }

    public void setClassifyNum(int classifyNum) {
        this.classifyNum = classifyNum;
    }

    public int getEntityNum() {
        return entityNum;
    }

    public void setEntityNum(int entityNum) {
        this.entityNum = entityNum;
    }

    public List<ClassifyEx> getClassifyExList() {
        return classifyExList;
    }

    public void setClassifyExList(List<ClassifyEx> classifyExList) {
        this.classifyExList = classifyExList;
    }
}
